package controller;

import java.io.IOException;
import java.util.Calendar;
import java.util.Date;
import java.util.StringTokenizer;

import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import models.Whois;

import com.google.appengine.api.datastore.Text;
import com.googlecode.objectify.Key;
import com.googlecode.objectify.Objectify;
import com.googlecode.objectify.ObjectifyService;

public class WhoisServlet extends HttpServlet {
	public void doGet(HttpServletRequest req, HttpServletResponse resp)
			throws IOException {
		
		resp.setCharacterEncoding("utf-8");
		String path = ((HttpServletRequest)req).getRequestURI();
		StringTokenizer st = new StringTokenizer( path,"/");
        int count = st.countTokens(); 

        if(count != 2)
        {
        	try {
				req.getRequestDispatcher("/404error.jsp").forward(req, resp);
			} catch (Exception e) {
				e.printStackTrace();
			}
        	return;
        }
        
		// skip one token /question/abcd (remove sites)
		st.nextToken();
		String title_url = st.nextToken();
		title_url = Utils.fomatDomain(title_url);
		if(title_url!=null&&Utils.checkDomain(title_url))
		{
			resp.setHeader( "Cache-Control", "public, max-age=86400");
			resp.setHeader( "Expires", new Date(Calendar.getInstance().getTime().getTime()+24*60*60*1000).toGMTString());
			try {
				ObjectifyService.register(Whois.class);
			} catch (Exception e) {
				// TODO: handle exception
			}
			Objectify objectify = ObjectifyService.begin();
			Whois whois = new Whois();
			whois.setDomain(title_url);
			try {
				whois = objectify.get(new Key<Whois>(Whois.class, title_url));
	
			} catch (Exception e) {
				whois.setWhois(new Text(server.Whois.GetWhois(title_url)));
				objectify.put(whois);
			}
			
			req.setAttribute("whois", whois);
			
			try {
				req.getRequestDispatcher("/whois.jsp").forward(req, resp);
			} catch (Exception e) {
				e.printStackTrace();
			}
		} else {
			resp.sendRedirect("/gsearch?cx=003091030381474148674:dwcpiom6xau&cof=FORID%3A10&ie=UTF-8&q="+title_url+"&sa=Search");
		}
	}
}
